import os
import time
import unittest
from HTMLTestRunner import HTMLTestRunner

# TODO loader 测试脚本加载
loader = unittest.defaultTestLoader.discover('./test', pattern='test*')

# TODO 准备测试报告文件
# 文件名规则：年月日时分秒-login.html
report = os.path.abspath('report')  # 存放报告文件目录的绝对路径
filename = time.strftime('%Y%m%d%H%M%S') + 'demo1.html' # 文件名
filepath = os.path.join(report, filename) # 报告文件路径

# TODO 操作配置测试报告信息，并返回一个runner
# 普通操作
'''
# 打开文件
f = open(filepath, mode='wb')
runner = HTMLTestRunner(
    title='测试报告标题',
    description='测试报告信息',
    tester='myself',  # 测试者
    stream=f,
    verbosity=2
)
# 关闭文件
# f.close()
'''
# with open
with open(filepath, 'wb') as f:
    runner = HTMLTestRunner(
        title='测试报告标题2',
        description='测试报告信息2',
        tester='myself',  # 测试者
        stream=f,
        verbosity=2
    )
    runner.run(loader)


